from PCR_Jittor.jittor.modules.sinkhorn.learnable_sinkhorn import LearnableLogOptimalTransport
